package cn.cxq.learning.recursion;

import java.util.ArrayList;
import java.util.List;

public class EightQueue {

    private int[] plan;
    private int maxSize;
    private List<int[]> plans;
    private int planCount;
    private int checkCount;

    public EightQueue() {
        this.plan = new int[8];
        this.maxSize = 8;
        this.plans = new ArrayList<>();
        this.planCount = 0;
        this.checkCount = 0;
    }

    public int getPlanCount() {
        return this.planCount;
    }

    /**
     * 开始放置并返回所有方案
     * @return
     */
    public List<int[]> startAndGetPlans(){
        this.setQueue(0);
        return this.plans;
    }

    /**
     * 放置第n个皇后
     * @param n
     */
    public void setQueue(int n){
        if (n == maxSize) {
            //print();
            // 在集合中存的只是一个指针，如果一直存this.plan的话，那最终数组里面接收到的是92个指针指向同一个数组，
            // 因为这个时候遍历结束了，所以我们会发现，当我们遍历this.plans的时候全部的值都是7，即7 7 7 7 7 7 7 7
            // 解决方案：使用数组的clone方法，分配新的内存空间，使用新的指针来指向它，就不会出现问题了
            //int[] clone = this.plan.clone();
            this.plans.add(this.plan.clone());
            planCount ++;
            return;
        }

        for (int i = 0; i < maxSize; i++) {

            this.plan[n] = i;
            if (check(n)) setQueue(n+1);
        }

    }

    private void print(){
        for (int i = 0; i < plan.length; i++) {
            System.out.print(plan[i] + " ");
        }
        System.out.println();
    }

    /**
     * 检查放置是否合法
     * @param n
     * @return
     */
    private boolean check(int n){

        checkCount ++;

        for (int i = 0; i < n; i++) {
            if (plan[n] == plan[i] || Math.abs(n-i) == Math.abs(plan[n]-plan[i])){
                return false;
            }
        }
        return true;
    }

    public int getCheckCount() {
        return checkCount;
    }
}
